Skip to content

feat(spark): connector Java SPI module (3/6)#106

Draft
timsaucer wants to merge 3 commits into
apache:mainfrom
timsaucer:split/03-spark-java-spi
Draft

feat(spark): connector Java SPI module (3/6)#106
timsaucer wants to merge 3 commits into
apache:mainfrom
timsaucer:split/03-spark-java-spi

Conversation

@timsaucer

@timsaucer timsaucer commented Jun 12, 2026

Copy link
Copy Markdown
Member

Stacked PR series (6 parts) — splitting the Spark DataSource V2 connector.
All six target main. They build on each other, so review and merge in order — until the earlier parts merge, this PR's diff includes their changes too.

  1. build: Cargo workspace + native-common extraction (1/6) #104 — Cargo workspace + native-common extraction
  2. feat(spark): datafusion-spark-bridge Rust SDK (2/6) #105datafusion-spark-bridge Rust SDK
    ➤ 3. feat(spark): connector Java SPI module (3/6) #106 — Spark connector Java SPI
  3. feat(spark): DataSource V2 connector, Scala (4/6) #107 — Spark DataSource V2 connector (Scala)
  4. feat(spark): bridge scaffold generator (5/6) #108 — Bridge scaffold generator
  5. feat(examples): end-to-end Spark bridge demo (6/6) #109 — End-to-end examples

Purpose

Introduce the spark Maven module and the pure-Java SPI a bridge implements: provider factory, scan backend, native library loader, options codec, and the partition/partitioning descriptors. The stable contracts the Scala connector and example bridges are written against.

Compiles standalone (no Scala yet); includes the SPI-only unit tests.

🤖 Generated with Claude Code

timsaucer and others added 3 commits June 12, 2026 13:23
…e-common

Move the standalone `native` crate into a root Cargo workspace and extract
shared JNI plumbing (error->exception mapping, Tokio runtime singleton,
StreamingReader) into a new `datafusion-jni-common` crate under `native-common/`.
`native/src/errors.rs` moves to `native-common/src/errors.rs`; the nine native
modules now import error/runtime helpers from `datafusion_jni_common`.

Build glue follows: single root `Cargo.lock`, `.cargo/config.toml` redirects
output to `rust-target/`, Makefile/CI/poms updated to build `--workspace` and
target `-p datafusion-jni`. Core javadoc build commands updated to match.

Pure refactor; no behavior change. First of a 6-PR stack splitting the Spark
DataSource V2 connector work.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
New `spark/bridge` workspace crate providing the `export_bridge!` macro that
generates the six JNI entry points a Spark connector bridge exposes
(providerSchemaIpc, createScan, partitionCount, executeStreamPartition,
executeStream, closeScan). Includes the options decoder, scan planning/execution
glue, and the Arrow type-widening layer (wraps any TableProvider for Spark type
compatibility).

Self-contained SDK with no Java/Scala coupling. Depends only on
datafusion-jni-common. Second of the 6-PR connector stack.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Introduce the `spark` Maven module and the pure-Java contracts a bridge
implements: BridgeProviderFactory (no-arg factory + scanBackend()), ScanBackend
(delegates to the bridge's JNI methods), NativeLibraryLoader (cdylib
extraction/loading), OptionsCodec (cross-language options encoder), PartitionInfo
(one entry per Spark task), and ReportedPartitioning (optional shuffle-elision
declaration).

Compiles standalone with no Scala main yet. Includes the two SPI-only tests
(OptionsCodecTest, BridgeProviderFactoryDefaultsTest). Third of the 6-PR stack.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
@timsaucer

Copy link
Copy Markdown
Member Author

Repo-layout docs follow-up (from #104 review): #104 trimmed docs/source/contributor-guide/development.md to the foundation crates only. This PR adds spark/pom.xml (the spark module) and the Java SPI, so re-add their doc slice when it lands:

  • add spark to the pom.xml modules bullet (currently core + examples);
  • add a spark/ bullet for the connector (Java SPI at this stage).

timsaucer added a commit to timsaucer/datafusion-java that referenced this pull request Jun 12, 2026
Address review feedback on the workspace-foundation PR:

- development.md: trim the repo-layout section to the crates this PR
  actually ships (native, native-common). It was forward-referencing
  spark/, spark/bridge, datafusion-spark-bridge, and examples/native --
  none of which exist until later PRs in the stack -- and called the
  member list "three" while listing four. Later PRs (apache#105/apache#106/apache#107/apache#109)
  carry notes to re-add their own slice when those dirs land.

- rat_exclude_files.txt: the Rust lockfile moved to the workspace root,
  so the stale native/Cargo.lock entry left the root Cargo.lock with no
  RAT exclude for the source-tarball check (check-rat-report.py). Point
  it at Cargo.lock.

- native-common: dedupe the panic-payload downcast -- StreamingReader::next
  now calls errors::panic_message instead of repeating the String/&str
  match inline.

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant